import torch
from torch import nn


class MyNN(nn.Module):
    def __init__(self):
        super(MyNN, self).__init__()

    def forward(self, input):
        output =input+1
        return output

my_nn = MyNN()
x=torch.tensor(1.0)
print(x)
x = my_nn(x)
print(x)
